# coding=utf8
from models.user.info import model_user_info
from Crypto.Hash import SHA512


def main():
    login = input("输入账号:")
    user = model_user_info.coll.find_one({
        'login': login
    })
    if user is None:  # 新增
        passwd = input("您是新用户,请设置密码:")
        user_name = input("输入用户名:")
        while True:
            appid = input("appid:")
            if model_user_info.coll.find_one({
                'WxAccounts': {
                    '$elemMatch': {
                        'appid': appid
                    }
                }
            }) is not None:
                print("appid已经存在了,请重新输入")
                continue
            appsecret = input("appsecret:")
            remark = input("备注:")
            break
        role = input("是否是管理员?(0为管理员1为普通用户)")
        model_user_info.coll.insert_one({
            'Login': login,
            'PassWd': SHA512.new(passwd.encode()).hexdigest(),
            'UserName': user_name,
            'Role': role,
            'WxAccounts': [
                dict(appid=appid, appsecret=appsecret, remark=remark)
            ]
        })
